from abc import ABC, abstractmethod
from ...pojo import ChatRequestDTO, ChatPlusRequestDTO, GeneratorImageDTO


class ChatService(ABC):
    @abstractmethod
    async def chat(self, chatRequest: ChatRequestDTO): ...

    @abstractmethod
    async def chatPlus(self, chatPlusRequest: ChatPlusRequestDTO): ...

    @abstractmethod
    async def generator_image(self, generatorImageDTO: GeneratorImageDTO): ...
